package hu.diploma.core.util;

import hu.diploma.common.vos.DeviceVo;
import hu.diploma.core.entity.Device;
import hu.diploma.core.entity.Group;
import hu.diploma.core.entity.Person;
import hu.diploma.core.entity.PersonalData;
import hu.diploma.core.entity.User;

import java.util.HashMap;

public class EntityVoMapper {

	public static HashMap<Class, Class> etityToVo = new HashMap<Class, Class>();
	public static HashMap<Class, Class> VOToEntity = new HashMap<Class, Class>();

	static {
		EntityVoMapper.etityToVo.put(Device.class, DeviceVo.class);
		EntityVoMapper.etityToVo.put(User.class, DeviceVo.class);
		EntityVoMapper.etityToVo.put(PersonalData.class, DeviceVo.class);
		EntityVoMapper.etityToVo.put(Group.class, DeviceVo.class);
		EntityVoMapper.etityToVo.put(Person.class, DeviceVo.class);
		EntityVoMapper.etityToVo.put(Person.class, DeviceVo.class);

		EntityVoMapper.VOToEntity.put(DeviceVo.class, Device.class);
	}

	public static HashMap<Class, Class> getEtityToVo() {
		return EntityVoMapper.etityToVo;
	}

	public static HashMap<Class, Class> getVOToEntity() {
		return EntityVoMapper.VOToEntity;
	}

}
